package cn.uce.common.dataSource;

/**
 * 数据源处理
 * @author Sunny
 * @date 2018-9-23
 */
public class DataSourceInvoke {
    /**
     * 切换数据源线程副本
     */
    private static final ThreadLocal<String> DATA_SOURCE_POOL= new ThreadLocal<String>();

    /**
     * 切换当前线程数据源
     * @param dataSource
     */
    public static void putDataSource(String dataSource){
        DATA_SOURCE_POOL.set(dataSource);
    }

    /**
     * 获取当前线程数据源
     * @return
     */
    public static String getDataSource(){
        return  DATA_SOURCE_POOL.get();
    }

    /**
     * 删除线程数据源
     */
    public static  void clearDataSource(){
        DATA_SOURCE_POOL.remove();
    }
}
